%%{init: {'theme': 'neutral'}}%%
flowchart TD
    A["push_chunk(chunk)"] --> B{chunk is None?}
    B -- Yes --> C[chunk = END_OF_STREAM]
    B -- No --> D{chunk is END_OF_STREAM?}
    D -- Yes --> E[pass – already sentinel]
    D -- No --> F{isinstance chunk, str?}
    F -- Yes --> G[valid string chunk]
    F -- No --> H[invalid type – ignored]
    C --> I["_process(END_OF_STREAM)"]
    E --> I
    G --> J["_process(chunk: str)"]
    I --> K{enable_buffer?}
    J --> K
    K -- Yes, str chunk --> L[buffer += chunk]
    K -- Yes, END_OF_STREAM --> M[noop / fall through]
    K -- No, str chunk --> N[completion += chunk, check stop tokens]
    K -- No, END_OF_STREAM --> O{pipe_to set?}
    N --> O
    O -- Yes --> P[asyncio.create_task pipe_to.push_chunk]
    O -- No, include_metadata --> Q[queue.put chunk_dict]
    O -- No, plain --> R[queue.put chunk]
    Q --> S{chunk is END_OF_STREAM?}
    R --> S
    S -- Yes --> T[streaming_finished_event.set]

%%{init: {'theme': 'base', 'themeVariables': {"darkMode": true, "background": "#0d1117", "primaryColor": "#21262d", "primaryTextColor": "#e6edf3", "primaryBorderColor": "#8b949e", "lineColor": "#8b949e", "textColor": "#e6edf3", "edgeLabelBackground": "#161b22", "actorBkg": "#21262d", "actorBorder": "#8b949e", "actorTextColor": "#e6edf3", "actorLineColor": "#8b949e", "signalColor": "#8b949e", "signalTextColor": "#e6edf3", "noteBkgColor": "#373320", "noteBorderColor": "#d4a72c", "noteTextColor": "#f0e6c0", "labelBoxBkgColor": "#21262d", "labelBoxBorderColor": "#8b949e", "labelTextColor": "#e6edf3", "loopTextColor": "#e6edf3", "activationBkgColor": "#30363d", "activationBorderColor": "#8b949e"}}}%%
flowchart TD
    A["push_chunk(chunk)"] --> B{chunk is None?}
    B -- Yes --> C[chunk = END_OF_STREAM]
    B -- No --> D{chunk is END_OF_STREAM?}
    D -- Yes --> E[pass – already sentinel]
    D -- No --> F{isinstance chunk, str?}
    F -- Yes --> G[valid string chunk]
    F -- No --> H[invalid type – ignored]
    C --> I["_process(END_OF_STREAM)"]
    E --> I
    G --> J["_process(chunk: str)"]
    I --> K{enable_buffer?}
    J --> K
    K -- Yes, str chunk --> L[buffer += chunk]
    K -- Yes, END_OF_STREAM --> M[noop / fall through]
    K -- No, str chunk --> N[completion += chunk, check stop tokens]
    K -- No, END_OF_STREAM --> O{pipe_to set?}
    N --> O
    O -- Yes --> P[asyncio.create_task pipe_to.push_chunk]
    O -- No, include_metadata --> Q[queue.put chunk_dict]
    O -- No, plain --> R[queue.put chunk]
    Q --> S{chunk is END_OF_STREAM?}
    R --> S
    S -- Yes --> T[streaming_finished_event.set]